#ifndef OBJECT_RECOGNITION_H_
#define OBJECT_RECOGNITION_H_

#include <opencv2/core/core.hpp>
#include <vector>
#include <string>

using namespace std;

namespace jf_obj_recog{

typedef void * HANDLE;
typedef std::pair<string, float> Prediction;

#ifdef __cplusplus
extern "C" {
#endif

	int object_recognition_init(HANDLE *pHandle,const string model_file,const string trained_file,const string mean_file,const vector<string> label_files,int GPUID);
	int object_recognition_process(HANDLE handle, cv::Mat img, vector<vector<Prediction> >& predict);
	int object_recognition_destory(HANDLE handle);

#ifdef __cplusplus
}
#endif

}

#endif  // OBJECT_RECOGNITION_H_
